


const tf = require('@tensorflow/tfjs-node-gpu');
async function matrixMulGPU(a, b) {
    // 创建 TensorFlow 张量
    const aTensor = tf.tensor(a);
    const bTensor = tf.tensor(b);

    // 使用 GPU 进行矩阵乘法
    const result = tf.matMul(aTensor, bTensor);

    // 等待结果完成
    await result.data();

    // 返回结果数组
    return result.arraySync();
}

const a = [[1, 2], [3, 4]]; // 你的矩阵 A
const b = [[5, 6], [7, 8]]; // 你的矩阵 B
for(let i=0;i<100;i++){
    const result =   matrixMulGPU(a, b);

    result.then(k=>{
        console.log(k);
    })
}



